tensor([[[[[0.0200, 0.0200],
           [0.0200, 0.0200]],

          [[0.5800, 0.6100],
           [0.6000, 0.5800]]],


         [[[0.5800, 0.5900],
           [0.5900, 0.5800]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0100, 0.0000],
           [0.0000, 0.0100]],

          [[0.4100, 0.3900],
           [0.4000, 0.4100]]],


         [[[0.4100, 0.4100],
           [0.4100, 0.4100]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[1.0000, 1.0000],
           [1.0000, 1.0000]]]],



        [[[[0.9600, 0.9700],
           [0.9800, 0.9600]],

          [[0.0100, 0.0000],
           [0.0000, 0.0100]]],


         [[[0.0100, 0.0000],
           [0.0000, 0.0100]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]]], grad_fn=<RoundBackward1>)